﻿using System.Collections;
using System.Collections.Generic;

namespace Mirror.FizzySteam
{
  public class BidirectionalDictionary<T1, T2> : IEnumerable
  {
    private Dictionary<T1, T2> t1ToT2Dict = new Dictionary<T1, T2>();
    private Dictionary<T2, T1> t2ToT1Dict = new Dictionary<T2, T1>();

    public IEnumerable<T1> FirstTypes => t1ToT2Dict.Keys;
    public IEnumerable<T2> SecondTypes => t2ToT1Dict.Keys;

    public IEnumerator GetEnumerator() => t1ToT2Dict.GetEnumerator();

    public int Count => t1ToT2Dict.Count;

    public void Add(T1 key, T2 value)
    {
      t1ToT2Dict[key] = value;
      t2ToT1Dict[value] = key;
    }

    public void Add(T2 key, T1 value)
    {
      t2ToT1Dict[key] = value;
      t1ToT2Dict[value] = key;
    }

    public T2 Get(T1 key) => t1ToT2Dict[key];

    public T1 Get(T2 key) => t2ToT1Dict[key];

    public bool TryGetValue(T1 key, out T2 value) => t1ToT2Dict.TryGetValue(key, out value);

    public bool TryGetValue(T2 key, out T1 value) => t2ToT1Dict.TryGetValue(key, out value);

    public bool Contains(T1 key) => t1ToT2Dict.ContainsKey(key);

    public bool Contains(T2 key) => t2ToT1Dict.ContainsKey(key);

    public void Remove(T1 key)
    {
      if (Contains(key))
      {
        T2 val = t1ToT2Dict[key];
        t1ToT2Dict.Remove(key);
        t2ToT1Dict.Remove(val);
      }
    }
    public void Remove(T2 key)
    {
      if (Contains(key))
      {
        T1 val = t2ToT1Dict[key];
        t1ToT2Dict.Remove(val);
        t2ToT1Dict.Remove(key);
      }
    }

    public T1 this[T2 key]
    {
      get => t2ToT1Dict[key];
      set
      {
        t2ToT1Dict[key] = value;
        t1ToT2Dict[value] = key;
      }
    }

    public T2 this[T1 key]
    {
      get => t1ToT2Dict[key];
      set
      {
        t1ToT2Dict[key] = value;
        t2ToT1Dict[value] = key;
      }
    }
  }
}